from paddlenlp import Taskflow

__information_extraction = Taskflow(
    "information_extraction",
    schema=['人物', '出生地', {'公司': ['地点', '成立时间']}],
    # home_path=NLP_HOME_PATH,
)
__text_summarization = Taskflow(
    "text_summarization",
    # home_path=NLP_HOME_PATH,
)


def information_extraction(input_data):
    output = __information_extraction(input_data)
    return output


def text_summarization(input_data):
    output = __text_summarization(input_data)
    return output
